# -*- coding: utf-8 -*-
"""
@Time    : 2024/7/19 11:50 
@Author  : ZhangShenao 
@File    : 5.使用HuggingFace远程Embedding模型.py 
@Desc    : 使用HuggingFaceEndpointEmbeddings,远程Embedding模型
"""
import dotenv
from langchain_huggingface import HuggingFaceEndpointEmbeddings

# 加载环境变量
dotenv.load_dotenv()

# 创建HuggingFaceEndpointEmbeddings,使用远程Embedding模型
embeddings = HuggingFaceEndpointEmbeddings(
    model='sentence-transformers/all-MiniLM-L12-v2',  # 指定模型名称
)

# 使用远程模型进行Embedding
vector = embeddings.embed_documents(texts=['床前明月光'])[0]
print(vector[:10])
print(len(vector))
